bool fun(int x) {
    if (x == 1)
        return false;
    for (int i = 2; i * i <= x; i++) {
        if (x % i == 0)
            return false;
    }
    return true;
}
class Solution {
public:
    int diagonalPrime(vector<vector<int>>& nums) {
        int ans = 0;
        int n = nums.size();
        int i = 0;
        for (int i = 0; i < n; i++) {
            if (fun(nums[i][i]))
                ans = max(nums[i][i], ans);
            if (fun(nums[i][n - i - 1]))
                ans = max(nums[i][n - i - 1], ans);
        }
        return ans;
    }
};